import torch


class MyModule(torch.nn.Module):
    def forward(self, x, y):
        return torch.add(x, y)



